import os # 载入os运行库
import shutil # 载入shutil运行库，这个库在本代码中的作用主要用于移动文件
file_dir = '未整理' # 想要整理的文件夹所在目录
move_to_dir = '整理后' # 整理后放到的文件夹
for root, dirs, files in os.walk(file_dir): # 遍历文件夹
    for f in files:
        ext = os.path.splitext(f)[1][1:] # 获取文件的扩展名
        print(ext) # 查看ext内容是否正确获得扩展名，确定没有问题
        path = os.path.join(root, f) # 获取文件的总路径
        if ext.lower() in ['jpg', 'jpeg', 'gif', 'png', 'bmp']: # 通过后缀名判断文件是否是图片
            shutil.move(path, move_to_dir) # 移动第一个文件到第二个文件
print('Done!')